package edu.dayu.demo3;

import java.util.HashMap;

/**
 * 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。
 * 输入: nums = [0,1,0]
 * 输出: 2
 * 说明: [0, 1] (或 [1, 0]) 是具有相同数量 0 和 1 的最长连续子数组。
 * 思路：首先，数组只含有 0，1 那么，0 和 1 的个数相同的话，总和会等于长度的一半。
 *      那怎么确定子数组？ 前缀和？ 记录 所有位置的和，数组下标记录了位置，然后从尾巴开始遍历。
 *      遍历的话，首先长度肯定是偶数。  时间复杂度  n^2 有点高，超时了。
 *      如果是将 sum 存入hashmap 里呢，而且在计算 sum 的时候，可以将 0 换做 -1.这样方便比较。
 *      只要 sum 中出现 0 ，或者 hashmap 后面出现和 sum 一样的 key，那就说明有符合题意的值。然后拿最大的即可。
 *
 * 思路 2：计算数组的和，判断和与长度的关系。如果小于等于长度的一半，直接 sum乘以 2，如果大于长度的一半，那就计算 0 个数的 乘以 2
 *        这样也不行，题目需要连续，这个无法保证连续。
 *
 */
public class 个数相同的子数组 {
    public static void main(String[] args) {
        int[] ints = {0,0,0,0,0,1,0};
        System.out.println(findMaxLength(ints));
    }

    /**
     * 前缀和
     * @param nums
     * @return
     */
    public static int findMaxLength(int[] nums) {
        int max = 0;
        int preSum = 0;
        HashMap<Integer, Integer> hashMap = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            preSum += nums[i] == 0 ? -1 :1;
            if (preSum == 0) max = i + 1;
            else if (hashMap.containsKey(preSum)) max = Math.max(max, i - hashMap.get(preSum));
            else hashMap.put(preSum, i);
        }
        return max;
    }

    /**
     * 方案 2
     * @param nums
     * @return
     */
    public static int findMaxLength2(int[] nums) {
        int len = nums.length;
        int sum = 0;
        for (int i = 0; i < len; i++) {
            sum+=nums[i];
        }
        return 2*sum<=len?2*sum:(len-sum)*2;
    }
}
